from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms
from PIL import Image
from cv2 import cv2
#python中的用法---tensor数据类型
#通过transforms.ToTensor解决两个问题:
#1.transforms应该如何使用
#2.为什么需要tensor数据类型:因为tensor数据类型包含了反向神经网络的一些理论参数

#绝对路径“E:\lcc\pytorch\learn_pytorch\dataset\train\ants_image\0013035.jpg
#相对路径”dataset/train/ants_image/0013035.jpg"
img_path = "dataset/train/ants_image/0013035.jpg"
img = Image.open(img_path)
# print(type(img))

writer = SummaryWriter("logs")

#1.transforms应该如何使用
tensor_trans = transforms.ToTensor()#从transfors中选择一个class进行创建自己的工具
tensor_img = tensor_trans(img)#根据创建使用的工具传入相应的参数
#print(tensor_img)
writer.add_image("Tensor_img",tensor_img)
writer.close()